home *** CD-ROM | disk | FTP | other *** search
/ AmigActive 2 / AACD 2.iso / AACD / Magazine / GraphicsCards / StormMesa / util / readtex.c < prev    next >
C/C++ Source or Header  |  1998-12-15  |  8KB  |  350 lines
  1. /* readtex.c */
  2.  
  3. /*
  4.  * Read an SGI .rgb image file and generate a mipmap texture set.
  5.  * Much of this code was borrowed from SGI's tk OpenGL toolkit.
  6.  */
  7.  
  8.  
  9.  
  10. #include <GL/gl.h>
  11. #include <GL/glu.h>
  12. #include <stdio.h>
  13. #include <stdlib.h> 
  14. #include <string.h>
  15.  
  16.  
  17. #ifndef SEEK_SET
  18. #  define SEEK_SET 0
  19. #endif
  20.  
  21.  
  22. /*
  23. ** RGB Image Structure
  24. */
  25.  
  26. typedef struct _TK_RGBImageRec {
  27.    GLint sizeX, sizeY;
  28.    GLint components;
  29.    unsigned char *data;
  30. } TK_RGBImageRec;
  31.  
  32.  
  33.  
  34. /******************************************************************************/
  35.  
  36. typedef struct _rawImageRec {
  37.     unsigned short imagic;
  38.     unsigned short type;
  39.     unsigned short dim;
  40.     unsigned short sizeX, sizeY, sizeZ;
  41.     unsigned long min, max;
  42.     unsigned long wasteBytes;
  43.     char name[80];
  44.     unsigned long colorMap;
  45.     FILE *file;
  46.     unsigned char *tmp, *tmpR, *tmpG, *tmpB, *tmpA;
  47.     unsigned long rleEnd;
  48.     GLuint *rowStart;
  49.     GLint *rowSize;
  50. } rawImageRec;
  51.  
  52. /******************************************************************************/
  53.  
  54. static void ConvertShort(unsigned short *array, long length)
  55. {
  56.    unsigned long b1, b2;
  57.    unsigned char *ptr;
  58.  
  59.    ptr = (unsigned char *)array;
  60.    while (length--) {
  61.       b1 = *ptr++;
  62.       b2 = *ptr++;
  63.       *array++ = (b1 << 8) | (b2);
  64.    }
  65. }
  66.  
  67. static void ConvertLong(GLuint *array, long length)
  68. {
  69.    unsigned long b1, b2, b3, b4;
  70.    unsigned char *ptr;
  71.  
  72.    ptr = (unsigned char *)array;
  73.    while (length--) {
  74.       b1 = *ptr++;
  75.       b2 = *ptr++;
  76.       b3 = *ptr++;
  77.       b4 = *ptr++;
  78.       *array++ = (b1 << 24) | (b2 << 16) | (b3 << 8) | (b4);
  79.    }
  80. }
  81.  
  82. static rawImageRec *RawImageOpen(const char *fileName)
  83. {
  84.    union {
  85.       int testWord;
  86.       char testByte[4];
  87.    } endianTest;
  88.    rawImageRec *raw;
  89.    GLenum swapFlag;
  90.    int x;
  91.  
  92.    endianTest.testWord = 1;
  93.    if (endianTest.testByte[0] == 1) {
  94.       swapFlag = GL_TRUE;
  95.    } else {
  96.       swapFlag = GL_FALSE;
  97.    }
  98.  
  99.    raw = (rawImageRec *)malloc(sizeof(rawImageRec));
  100.    if (raw == NULL) {
  101.       fprintf(stderr, "Out of memory!\n");
  102.       return NULL;
  103.    }
  104.    if ((raw->file = fopen(fileName, "rb")) == NULL) {
  105.       perror(fileName);
  106.       return NULL;
  107.    }
  108.  
  109.    fread(raw, 1, 12, raw->file);
  110.  
  111.    if (swapFlag) {
  112.       ConvertShort(&raw->imagic, 6);
  113.    }
  114.  
  115.    raw->tmp = (unsigned char *)malloc(raw->sizeX*256);
  116.    raw->tmpR = (unsigned char *)malloc(raw->sizeX*256);
  117.    raw->tmpG = (unsigned char *)malloc(raw->sizeX*256);
  118.    raw->tmpB = (unsigned char *)malloc(raw->sizeX*256);
  119.    if (raw->sizeZ==4) {
  120.       raw->tmpA = (unsigned char *)malloc(raw->sizeX*256);
  121.    }
  122.    if (raw->tmp == NULL || raw->tmpR == NULL || raw->tmpG == NULL ||
  123.        raw->tmpB == NULL) {
  124.       fprintf(stderr, "Out of memory!\n");
  125.       return NULL;
  126.    }
  127.  
  128.    if ((raw->type & 0xFF00) == 0x0100) {
  129.       x = raw->sizeY * raw->sizeZ * sizeof(GLuint);
  130.       raw->rowStart = (GLuint *)malloc(x);
  131.       raw->rowSize = (GLint *)malloc(x);
  132.       if (raw->rowStart == NULL || raw->rowSize == NULL) {
  133.      fprintf(stderr, "Out of memory!\n");
  134.      return NULL;
  135.       }
  136.       raw->rleEnd = 512 + (2 * x);
  137.       fseek(raw->file, 512, SEEK_SET);
  138.       fread(raw->rowStart, 1, x, raw->file);
  139.       fread(raw->rowSize, 1, x, raw->file);
  140.       if (swapFlag) {
  141.      ConvertLong(raw->rowStart, x/sizeof(GLuint));
  142.      ConvertLong((GLuint *)raw->rowSize, x/sizeof(GLint));
  143.       }
  144.    }
  145.    return raw;
  146. }
  147.  
  148. static void RawImageClose(rawImageRec *raw)
  149. {
  150.  
  151.    fclose(raw->file);
  152.    free(raw->tmp);
  153.    free(raw->tmpR);
  154.    free(raw->tmpG);
  155.    free(raw->tmpB);
  156.    if (raw->sizeZ>3) {
  157.       free(raw->tmpA);
  158.    }
  159.    free(raw);
  160. }
  161.  
  162. static void RawImageGetRow(rawImageRec *raw, unsigned char *buf, int y, int z)
  163. {
  164.    unsigned char *iPtr, *oPtr, pixel;
  165.    int count;
  166.  
  167.    if ((raw->type & 0xFF00) == 0x0100) {
  168.       fseek(raw->file, raw->rowStart[y+z*raw->sizeY], SEEK_SET);
  169.       fread(raw->tmp, 1, (unsigned int)raw->rowSize[y+z*raw->sizeY],
  170.         raw->file);
  171.       
  172.       iPtr = raw->tmp;
  173.       oPtr = buf;
  174.       while (1) {
  175.      pixel = *iPtr++;
  176.      count = (int)(pixel & 0x7F);
  177.      if (!count) {
  178.         return;
  179.      }
  180.      if (pixel & 0x80) {
  181.         while (count--) {
  182.            *oPtr++ = *iPtr++;
  183.         }
  184.      } else {
  185.         pixel = *iPtr++;
  186.         while (count--) {
  187.            *oPtr++ = pixel;
  188.         }
  189.      }
  190.       }
  191.    } else {
  192.       fseek(raw->file, 512+(y*raw->sizeX)+(z*raw->sizeX*raw->sizeY),
  193.         SEEK_SET);
  194.       fread(buf, 1, raw->sizeX, raw->file);
  195.    }
  196. }
  197.  
  198.  
  199. static void RawImageGetData(rawImageRec *raw, TK_RGBImageRec *final)
  200. {
  201.    unsigned char *ptr;
  202.    int i, j;
  203.  
  204.    final->data = (unsigned char *)malloc((raw->sizeX+1)*(raw->sizeY+1)*4);
  205.    if (final->data == NULL) {
  206.       fprintf(stderr, "Out of memory!\n");
  207.    }
  208.  
  209.    ptr = final->data;
  210.    for (i = 0; i < (int)(raw->sizeY); i++) {
  211.       RawImageGetRow(raw, raw->tmpR, i, 0);
  212.       RawImageGetRow(raw, raw->tmpG, i, 1);
  213.       RawImageGetRow(raw, raw->tmpB, i, 2);
  214.       if (raw->sizeZ>3) {
  215.      RawImageGetRow(raw, raw->tmpA, i, 3);
  216.       }
  217.       for (j = 0; j < (int)(raw->sizeX); j++) {
  218.      *ptr++ = *(raw->tmpR + j);
  219.      *ptr++ = *(raw->tmpG + j);
  220.      *ptr++ = *(raw->tmpB + j);
  221.      if (raw->sizeZ>3) {
  222.         *ptr++ = *(raw->tmpA + j);
  223.      }
  224.       }
  225.    }
  226. }
  227.  
  228.  
  229. static TK_RGBImageRec *tkRGBImageLoad(const char *fileName)
  230. {
  231.    rawImageRec *raw;
  232.    TK_RGBImageRec *final;
  233.  
  234.    if ((raw = RawImageOpen(fileName)) == NULL)
  235.     return NULL;
  236.    final = (TK_RGBImageRec *)malloc(sizeof(TK_RGBImageRec));
  237.    if (final == NULL) {
  238.       fprintf(stderr, "Out of memory!\n");
  239.       return NULL;
  240.    }
  241.    final->sizeX = raw->sizeX;
  242.    final->sizeY = raw->sizeY;
  243.    final->components = raw->sizeZ;
  244.    RawImageGetData(raw, final);
  245.    RawImageClose(raw);
  246.    return final;
  247. }
  248.  
  249.  
  250. static void FreeImage( TK_RGBImageRec *image )
  251. {
  252.    free(image->data);
  253.    free(image);
  254. }
  255.  
  256.  
  257. /*
  258.  * Load an SGI .rgb file and generate a set of 2-D mipmaps from it.
  259.  * Input:  imageFile - name of .rgb to read
  260.  *         intFormat - internal texture format to use, or number of components
  261.  * Return:  GL_TRUE if success, GL_FALSE if error.
  262.  */
  263. GLboolean LoadRGBMipmaps( const char *imageFile, GLint intFormat )
  264. {
  265.    GLint error;
  266.    GLenum format;
  267.    TK_RGBImageRec *image;
  268.  
  269.    image = tkRGBImageLoad( imageFile );
  270.    if (!image) {
  271.       return GL_FALSE;
  272.    }
  273.  
  274.    if (image->components==3) {
  275.       format = GL_RGB;
  276.    }
  277.    else if (image->components==4) {
  278.       format = GL_RGBA;
  279.    }
  280.    else {
  281.       /* not implemented */
  282.       fprintf(stderr,
  283.           "Error in LoadRGBMipmaps %d-component images not implemented\n",
  284.           image->components );
  285.       return GL_FALSE;
  286.    }
  287.  
  288.    error = gluBuild2DMipmaps( GL_TEXTURE_2D,
  289.                   intFormat,
  290.                   image->sizeX, image->sizeY,
  291.                   format,
  292.                   GL_UNSIGNED_BYTE,
  293.                   image->data );
  294.  
  295.    FreeImage(image);
  296.    return error ? GL_FALSE : GL_TRUE;
  297. }
  298.  
  299.  
  300.  
  301. /*
  302.  * Load an SGI .rgb file and return a pointer to the image data.
  303.  * Input:  imageFile - name of .rgb to read
  304.  * Output:  width - width of image
  305.  *          height - height of image
  306.  *          format - format of image (GL_RGB or GL_RGBA)
  307.  * Return:  pointer to image data or NULL if error
  308.  */
  309. GLubyte *LoadRGBImage( const char *imageFile, GLint *width, GLint *height,
  310.                GLenum *format )
  311. {
  312.    TK_RGBImageRec *image;
  313.    GLint bytes;
  314.    GLubyte *buffer;
  315.  
  316.    image = tkRGBImageLoad( imageFile );
  317.    if (!image) {
  318.       return NULL;
  319.    }
  320.  
  321.    if (image->components==3) {
  322.       *format = GL_RGB;
  323.    }
  324.    else if (image->components==4) {
  325.       *format = GL_RGBA;
  326.    }
  327.    else {
  328.       /* not implemented */
  329.       fprintf(stderr,
  330.           "Error in LoadRGBImage %d-component images not implemented\n",
  331.           image->components );
  332.       return NULL;
  333.    }
  334.  
  335.    *width = image->sizeX;
  336.    *height = image->sizeY;
  337.  
  338.    bytes = image->sizeX * image->sizeY * image->components;
  339.    buffer = (GLubyte *) malloc(bytes);
  340.    if (!buffer)
  341.       return NULL;
  342.  
  343.    memcpy( (void *) buffer, (void *) image->data, bytes );
  344.  
  345.    FreeImage(image);
  346.  
  347.    return buffer;
  348. }
  349.  
  350.